#pragma once
#include "src/common/common.h"

template<typename T>
__global__ void cuda_matmul(T* A,T* B,T* C,int rows,int cols,int total_num);


template<>
__global__ void cuda_matmul(half* A,half* B,half* C,int rows,int cols,int total_num);


template<typename T>
void launch_cuda_matmul(T* A,T* B,T* C,int rows,int cols,int blocksize,int total_nums);


template<typename T>
void launch_cpu_matmul(T* A,T* B,T* C,int rows,int cols);

template<>
void launch_cpu_matmul(half* A,half* B,half* C,int rows,int cols);